package com.ruoyi.system.utils;

import com.ruoyi.common.utils.StringUtils;
import lombok.Getter;
import lombok.Setter;
import org.apache.poi.hssf.usermodel.HeaderFooter;
import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.*;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.util.List;
import java.util.Map;


public class ExcelUtil {
    /**
     * @param response
     * @param name     文件名
     * @param title    表格标题名称(第一行)
     * @param header   表头(第二行)
     * @param datas    表格内容 List<List<Object>>
     * @throws IOException
     */
    public static void generateExcel(HttpServletResponse response, String name, String title, String[] header, List<Map> datas) throws IOException {
        response.setContentType("application/binary;charset=UTF-8");
//
        try {
            try {
                response.reset(); // 必要地清除response中的缓存信息
                response.setHeader("Access-Control-Allow-Origin","*");
                response.setHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode(name + ".xlsx", "UTF-8"));

            } catch (UnsupportedEncodingException e1) {
                e1.printStackTrace();
            }

            //创建工作薄
            XSSFWorkbook workbook = new XSSFWorkbook();
            //创建表单
            XSSFSheet sheet = genSheet(workbook, "testExcel");
            //创建表单样式
            XSSFCellStyle titleStyle = genTitleStyle(workbook);//创建标题样式
            XSSFCellStyle contextStyle = genContextStyle(workbook);//创建文本样式

            //创建Excel
            genExcel(sheet, titleStyle, contextStyle, title, header, datas);

            //将生成的Excel文件保存到本地
/*        FileOutputStream out = new FileOutputStream(new File("C:/Users/Administrator/Desktop/text/test.xlsx"));

            //将工作薄写入文件输出流中
            workbook.write(out);
            //文本文件输出流，释放资源
            out.flush();
            out.close();*/
            // 第七步，将文件输出到客户端浏览器
            try {
                OutputStream out = response.getOutputStream();
                workbook.write(out);
                out.flush();
                out.close();
            } catch (Exception e) {
                e.printStackTrace();
                System.out.println(e);
            }

            System.out.println("cg");
//            return "success";
        } catch (Exception e) {
            e.printStackTrace();
//            return "导出信息失败";
            System.out.println("sb");
        }
    }

    public static void genExcel(XSSFSheet sheet, XSSFCellStyle titleStyle, XSSFCellStyle contextStyle, String title, String[] header, List<Map> datas) {

        //根据Excel列名长度，指定列名宽度  Excel总共10列
        for (int i = 0; i < header.length; i++) {
            sheet.autoSizeColumn(i);
            sheet.setColumnWidth(i, 1500);
            sheet.setColumnWidth(i, sheet.getColumnWidth(i) * 30 / header.length);

        }

        //设置标题位置
        sheet.addMergedRegion(new CellRangeAddress(
                0, //first row
                0, //last row
                0, //first column
                9 //last column
        ));

        XSSFRow row = sheet.createRow(0);//创建第一行，为标题，index从0开始
        XSSFCell cell;
        cell = row.createCell(0);//创建一列
        cell.setCellValue(title);//标题
        cell.setCellStyle(titleStyle);//设置标题样式

        row = sheet.createRow(1);//创建第二行
        for (int i = 0; i < header.length; i++) {
            cell = row.createCell(i);
//            cell.setCellValue(header[i]);
//            cell.setCellStyle(contextStyle);
            if (StringUtils.isNotEmpty(header[i])) {
                cell.setCellValue(header[i]);
                cell.setCellStyle(contextStyle);
            } else {
                cell.setCellValue(" ");
                cell.setCellStyle(contextStyle);

            }
        }

        /*XSSFRow row = sheet.createRow(0);//创建第一行，为标题，index从0开始
        XSSFCell cell;
        cell = row.createCell(0);//创建一列
        cell.setCellValue("xxx 幼儿园一年级二班学生信息");//标题
        cell.setCellStyle(titleStyle);//设置标题样式

        row = sheet.createRow(1);//创建第二行
        cell = row.createCell(0);//创建第二行第一列
        cell.setCellValue("姓名");//第二行第一列内容
        cell.setCellStyle(contextStyle);//设置样式
        cell = row.createCell(1);//2 row 1 column
        cell.setCellValue("性别");
        cell.setCellStyle(contextStyle);
        cell = row.createCell(2);//2 row 2 column
        cell.setCellValue("学号");
        cell.setCellStyle(contextStyle);
        cell = row.createCell(3);//2 row 3 column
        cell.setCellValue("出生年月");
        cell.setCellStyle(contextStyle);
        cell = row.createCell(4);//2 row 4 column
        cell.setCellValue("家庭住址");
        cell.setCellStyle(contextStyle);
        cell = row.createCell(5);//2 row 5 column
        cell.setCellValue("语文");
        cell.setCellStyle(contextStyle);
        cell = row.createCell(6);//2 row 6 column
        cell.setCellValue("英语");
        cell.setCellStyle(contextStyle);
        cell = row.createCell(7);//2 row 7 column
        cell.setCellValue("数学");
        cell.setCellStyle(contextStyle);
        cell = row.createCell(8);//2 row 8 column
        cell.setCellValue("自然");
        cell.setCellStyle(contextStyle);
        cell = row.createCell(9);//2 row 9 column
        cell.setCellValue("科学");
        cell.setCellStyle(contextStyle);*/

		/*
			实际填充数据的时候，对可能为空的数据要进行处理，要先进行判断，否则报错
			String value = Test.getVal();
			if(StringUtils.isNotBlank(value)) {
				cell.setCellValue(value);
			} else{
			    cell.setCellValue(" ");
			}
		*/
        //从数据库取数据填充到Excel，这步省略，添加模拟数据
        //创建内容
        if (datas!=null&&datas.size() > 0) {
            for (int i = 0; i < datas.size(); i++) {
                row = sheet.createRow(i + 2);
                Map map = datas.get(i);
                int j = 0;
                for (Object value : map.values()) {
                    //将内容按顺序赋给对应的列对象
                    if (StringUtils.isNotEmpty(value.toString())) {
                        row.createCell(j).setCellValue(value.toString());
                        cell.setCellStyle(contextStyle);
                    } else {
                        row.createCell(j).setCellValue(" ");
                        cell.setCellStyle(contextStyle);
                    }
                    j++;

                }

            }
        }

       /* for(int i = 2 ; i<15;i++){//i从2开始计数，因为上面已经创建了 0 1行
            row=sheet.createRow(i);//创建第三行
            cell=row.createCell(0);//创建第三行第一列
            cell.setCellValue("小明");//第三行第一列的值
            cell.setCellStyle(contextStyle);//设置样式
            cell=row.createCell(1);//2 row 1 column
            cell.setCellValue("男");
            cell.setCellStyle(contextStyle);
            cell=row.createCell(2);//2 row 2 column
            cell.setCellValue("NO20180901"+i);
            cell.setCellStyle(contextStyle);
            cell=row.createCell(3);//2 row 3 column
            cell.setCellValue("2000/02/01");
            cell.setCellStyle(contextStyle);
            cell=row.createCell(4);//2 row 4 column
            cell.setCellValue("成都市天府广场");
            cell.setCellStyle(contextStyle);
            cell=row.createCell(5);//2 row 5 column
            cell.setCellValue("90");
            cell.setCellStyle(contextStyle);
            cell=row.createCell(6);//2 row 6 column
            cell.setCellValue("90");
            cell.setCellStyle(contextStyle);
            cell=row.createCell(7);//2 row 7 column
            cell.setCellValue("90");
            cell.setCellStyle(contextStyle);
            cell=row.createCell(8);//2 row 8 column
            cell.setCellValue("98");
            cell.setCellStyle(contextStyle);
            cell=row.createCell(9);//2 row 9 column
            cell.setCellValue("100");
            cell.setCellStyle(contextStyle);
        }*/
    }
    @Getter
    @Setter
    public class ExcelExp {

        private String fileName;// sheet的名称
        private String[] handers;// sheet里的标题
        private List<String[]> dataset;// sheet里的数据集

        public ExcelExp(String fileName, String[] handers, List<String[]> dataset) {
            this.fileName = fileName;
            this.handers = handers;
            this.dataset = dataset;
        }

        //getter、setter方法

    }
    /**
     *
     * @Title: exportManySheetExcel
     * @Description: 可生成单个、多个sheet
     * @param @param file 导出文件路径
     * @param @param mysheets
     * @return void
     * @throws
     */
    public static void exportManySheetExcel(String file, List<ExcelExp> mysheets){

        HSSFWorkbook wb = new HSSFWorkbook();//创建工作薄
        List<ExcelExp> sheets = mysheets;

        //表头样式
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HorizontalAlignment.CENTER); // 创建一个居中格式
        //字体样式
        HSSFFont fontStyle = wb.createFont();
        fontStyle.setFontName("微软雅黑");
        fontStyle.setFontHeightInPoints((short)12);

        style.setFont(fontStyle);

        for(ExcelExp excel: sheets){
            //新建一个sheet
            HSSFSheet sheet = wb.createSheet(excel.getFileName());//获取该sheet名称

            String[] handers = excel.getHanders();//获取sheet的标题名
            HSSFRow rowFirst = sheet.createRow(0);//第一个sheet的第一行为标题
            //写标题
            for(int i=0;i<handers.length;i++){
                //获取第一行的每个单元格
                HSSFCell cell = rowFirst.createCell(i);
                //往单元格里写数据
                cell.setCellValue(handers[i]);
                cell.setCellStyle(style); //加样式
                sheet.setColumnWidth(i, 4000); //设置每列的列宽
            }

            //写数据集
            List<String[]> dataset = excel.getDataset();
            for(int i=0;i<dataset.size();i++){
                String[] data = dataset.get(i);//获取该对象

                //创建数据行
                HSSFRow row = sheet.createRow(i+1);

                for(int j=0;j<data.length;j++){
                    //设置对应单元格的值
                    row.createCell(j).setCellValue(data[j]);
                }
            }
        }
        // 写文件
        try {

            FileOutputStream out = new FileOutputStream(new File(file));
            out.flush();
            wb.write(out);
            out.close();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    //设置表单，并生成表单
    public static XSSFSheet genSheet(XSSFWorkbook workbook, String sheetName) {
        //生成表单
        XSSFSheet sheet = workbook.createSheet(sheetName);
        //设置表单文本居中
        sheet.setHorizontallyCenter(true);
        sheet.setFitToPage(false);
        //打印时在底部右边显示文本页信息
        Footer footer = sheet.getFooter();
        footer.setRight("Page " + HeaderFooter.numPages() + " Of " + HeaderFooter.page());
        //打印时在头部右边显示Excel创建日期信息
        Header header = sheet.getHeader();
        header.setRight("Create Date " + HeaderFooter.date() + " " + HeaderFooter.time());
        //设置打印方式
        XSSFPrintSetup ps = sheet.getPrintSetup();
        ps.setLandscape(true); // true：横向打印，false：竖向打印 ，因为列数较多，推荐在打印时横向打印
//        ps.setPaperSize(HSSFPrintSetup.A4_PAPERSIZE); //打印尺寸大小设置为A4纸大小
        return sheet;
    }

    //创建文本样式
    public static XSSFCellStyle genContextStyle(XSSFWorkbook workbook) {
        XSSFCellStyle style = workbook.createCellStyle();
        style.setAlignment(HorizontalAlignment.CENTER);//文本水平居中显示
        style.setVerticalAlignment(VerticalAlignment.CENTER);//文本竖直居中显示
        style.setWrapText(true);//文本自动换行

        //生成Excel表单，需要给文本添加边框样式和颜色
        /*
             CellStyle.BORDER_DOUBLE      双边线
             CellStyle.BORDER_THIN        细边线
             CellStyle.BORDER_MEDIUM      中等边线
             CellStyle.BORDER_DASHED      虚线边线
             CellStyle.BORDER_HAIR        小圆点虚线边线
             CellStyle.BORDER_THICK       粗边线
         */
        style.setBorderBottom(BorderStyle.THIN);//设置文本边框
        style.setBorderLeft(BorderStyle.THIN);
        style.setBorderRight(BorderStyle.THIN);
        style.setBorderTop(BorderStyle.THIN);
        //set border color
//        style.setTopBorderColor(new XSSFColor(Color.BLACK));//设置文本边框颜色
//        style.setBottomBorderColor(new XSSFColor(Color.BLACK));
//        style.setLeftBorderColor(new XSSFColor(Color.BLACK));
//        style.setRightBorderColor(new XSSFColor(Color.BLACK));

        return style;
    }

    //生成标题样式
    public static XSSFCellStyle genTitleStyle(XSSFWorkbook workbook) {

        XSSFCellStyle style = workbook.createCellStyle();
        style.setAlignment(HorizontalAlignment.CENTER);
        style.setVerticalAlignment(VerticalAlignment.CENTER);
        style.setWrapText(true);


        //标题居中，没有边框，所以这里没有设置边框，设置标题文字样式
        XSSFFont titleFont = workbook.createFont();
        titleFont.setBold(true);//加粗
        titleFont.setFontHeight((short) 15);//文字尺寸
        titleFont.setFontHeightInPoints((short) 15);
        style.setFont(titleFont);

        return style;
    }

}

